import asyncio
import time
import ygg_aiomysql as aiomysql

async def main():
    """
    示例07: 检查数据库存在性
    """
    # 使用时间戳生成唯一数据库名
    test_db_name = f"test_db_{int(time.time())}"

    # 创建临时连接池
    pool = await aiomysql.new_pool(password="zhangdapeng520")

    # 检查数据库不存在
    print(f"检查数据库 {test_db_name} 是否存在: {await  aiomysql.has_database(pool, test_db_name)}")

    # 创建测试数据库
    await  aiomysql.add_database(pool, test_db_name)
    
    # 再次检查数据库存在
    print(f"创建后检查数据库 {test_db_name} 是否存在: {await  aiomysql.has_database(pool, test_db_name)}")

    # 删除测试数据库
    await  aiomysql.delete_database(pool, test_db_name)

    # 最终检查数据库不存在
    print(f"删除后检查数据库 {test_db_name} 是否存在: {await  aiomysql.has_database(pool, test_db_name)}")

    # 关闭连接池
    pool.close()
    await pool.wait_closed()

if __name__ == "__main__":
    asyncio.run(main())